"""
https://www.mairui.club/hsdata.html

所属指数
API接口：http://api.mairui.club/hscp/sszs/股票代码(如000001)/您的licence
备用接口：http://api1.mairui.club/hscp/sszs/股票代码(如000001)/您的licence
包年版接口：专用服务器(API接口前缀为 http://y.mairui.club)
钻石版接口：专用服务器(API接口前缀为 http://v.mairui.club)
接口说明：根据《股票列表》得到的股票代码获取上市公司的所属指数。
数据更新：每天15:30开始更新，次日凌晨3点前完成
请求频率：1分钟20次 | 包年版1分钟3千次 | 钻石版1分钟6千次
返回格式：标准Json格式      [{},...{}]
"""

import requests
from dao.company_index_dao import CompanyIndex
from api.stock_info import mr_licence

def get_stock_index(stock_code):
    api_url = "http://api.mairui.club/hscp/sszs/{0}/{1}".format(stock_code, mr_licence)
    response = requests.get(api_url)
    if response.status_code == 200:
        return response.json()
    else:
        return None


if __name__ == "__main__":
    stock_code = "000001"
    result = get_stock_index(stock_code)
    print(result)
    CompanyIndex.insert_or_update(stock_code, result)
    print(result)
